how to translate from matlab to python